#include "stl.h"
#include <random>
#include "solution.h"

int main(int argc, char *argv[]){
    struct {
        int n;
        vector<int> weights;
    } testcases[] = {
        // {1, {1}},
        {5, {1, 3}},
    };
    for(auto & ts : testcases){
        print("weights=", ts.weights, "\n")
        Solution sln(ts.weights);
        vector<int> ans;
        for(int i=0; i<ts.n; i++)
            ans.push_back(sln.pickIndex());
        print("ans:", ans, "\n")
    }
    
    print("\n------ main exit ------\n")
    return 0;
}
